Skip to content

Inference Code#15

Open
hamzakhalil798 wants to merge 7 commits intoVitjanZ:mainfrom
hamzakhalil798:main
Open

Inference Code#15
hamzakhalil798 wants to merge 7 commits intoVitjanZ:mainfrom
hamzakhalil798:main

Conversation

@hamzakhalil798
Copy link
Copy Markdown

Hey! I've added the Inference part to the code.
Changes :
Added a new class in dataloader.py for visualization. It loads two images instead of All images in the dataset.
Added visualize_DRAEM.py, it's basically the same as test_DRAEM.py with minor changes to display Original Images and Resulting Anomaly Heatmaps.

@adbensi
Copy link
Copy Markdown

adbensi commented Aug 20, 2023

[hamzakhalil798] Please, wouldn't it be more common to just upload an image for inference rather than uploading files from the training folder itself?

MAwaisr and others added 2 commits March 2, 2024 11:43
chmod +x command is added for clarity to avoid permission denial while downloading datasets and pretrained models
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants